# To add a new cell, type '# %%'
# To add a new markdown cell, type '# %% [markdown]'
# %%
import sys
import pso_fs

# %%
if __name__ == "__main__":
    if len(sys.argv) < 4:
        raise Exception("Please input enough parameters: *.data, *.labels, dir_for_imgs.")        

    pso_trainer = pso_fs.PSO_FS(initialize_strategy="most_small_other_large", lbest=True, size=40)
    #pso_trainer.start("../datasets/Madelon/madelon_train.data", "../datasets/Madelon/madelon_train.labels")
    #pso_trainer.start("./datasets/Leukemia/Leukemia1.data", "./datasets/Leukemia/Leukemia1.labels")
    pso_trainer.start(sys.argv[1], sys.argv[2], sys.argv[3])

